package 剑指offer;


import java.util.HashMap;
import java.util.Map;

class Node {
    int val;
    Node next;
    Node random;

    Node left;
    Node right;

    public Node(int val) {
        this.val = val;
        this.next = null;
        this.random = null;
    }
}

public class p35复杂链表的复制 {
    public Node copyRandomList(Node head) {
        if (head==null) {
            return null;
        }

        Node cur=head;
        Map<Node,Node> map=new HashMap<>();

        //第一次遍历：根据师父造徒弟，即赋值链表中的节点
        while (cur!=null) {
            Node node=new Node(cur.val);
            map.put(cur,node);
            cur=cur.next;
        }

        //第二次遍历：根据师父之间的关系构建徒弟之间的关系，即处理 node的next、random指向
        cur=head;
        while (cur!=null) {
            map.get(cur).next=map.get(cur.next);
            map.get(cur).random=map.get(cur.random);
            cur=cur.next;
        }

        //返回 徒弟的头结点，即复制链表的头结点
        return map.get(head);
    }
}
